-
Notifications
You must be signed in to change notification settings - Fork 13.4k
Add ARANGE Operator to SYCL Backend (Small & Focused Changes) #16362
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
This PR adds the @ggerganov @ggerganov/ggml-maintainers – |
Hi @Rbiessy and @NeoZhangJianyu, Following up on this PR after 3 weeks - adds ARANGE operator support for SYCL backend with significant performance improvements. The changes are minimal and fully tested. Would appreciate a review when possible. Thanks! |
Hello @GittyBurstein, sorry I am not maintaining the SYCL backend anymore. I'm not sure who else can help you merge this PR. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It's good work!
I test by UT. It's passed!
Thank you!
@GittyBurstein Thank you! |
96f5395
to
1c65a24
Compare
Hi @NeoZhangJianyu, Thank you very much for reviewing the code! Thanks again! |
Hi @NeoZhangJianyu, I forgot to mention earlier that two CI tests failed, but after reviewing the failures, I realized they were not related to my changes. Everything that was related to my code has been properly fixed. Thank you very much, and I would be happy to see this merged. 😉 |
Hi @CISC 👋 My PR for adding the ARANGE operator to the SYCL backend was already reviewed and approved by the SYCL maintainer, who confirmed that the implementation is correct and only required minor formatting fixes (extra spaces, etc.) for the CI checks. I’ve applied all the requested formatting corrections, but since this PR is from a fork, the CI workflows are currently awaiting approval. If everything looks good on your side, I’d really appreciate it if you could approve/run the workflows and merge the PR when convenient. |
Summary
Implements the
ARANGE
operator for the SYCL backend, generating evenly-spaced sequences on SYCL devices (Intel GPUs). Enables fast tensor initialization. The changes are small and focused.Changes
element_wise.cpp
ggml-sycl.cpp
Implementation
start
,step
, number of elements = tensor sizeTesting
Performance
Compatibility